package com.itany.corejava.code13_jdbc;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

/**
 * @author 石小俊
 * @date 2023年05月29日 14:11
 */
public class Test09_多表查询 {

    public static void main(String[] args) throws DataAccessException {
//        List<Emp> emps = selectAll();
//        for (Emp emp : emps) {
//            System.out.println(emp);
//        }

        Dept dept = selectDeptById(1);
        System.out.println(dept);
    }


    /**
     * 多对一查询
     * 查询所有员工以及对应员工所属的部门信息
     *
     * @return
     */
    public static List<Emp> selectAll() throws DataAccessException {
        Connection conn = null;
        PreparedStatement ps = null;
        ResultSet rs = null;
        List<Emp> emps = new ArrayList<>();
        try {
            conn = JDBCUtil.getConnection();
            String sql = new StringBuffer()
                    .append("select e.id,e.name,e.salary,d.id 'd.id',d.name 'd.name',d.remark ")
                    .append(" from t_emp e ")
                    .append(" left join t_dept d ")
                    .append("    on e.dept_id = d.id ")
                    .toString();
            ps = conn.prepareStatement(sql);
            rs = ps.executeQuery();
            while (rs.next()) {
                // 此时的映射涉及到两个对象
                // 但是Dept对象是作为Emp对象的属性存在的
                // 映射的本质是映射Emp对象
                Emp emp = new Emp();
                emp.setId(rs.getInt("id"));
                emp.setName(rs.getString("name"));
                emp.setSalary(rs.getDouble("salary"));

                // 准备一个Dept对象,接收结果集中的数据,进行映射
                Dept dept = new Dept();
                dept.setId(rs.getInt("d.id"));
                dept.setName(rs.getString("d.name"));
                dept.setRemark(rs.getString("remark"));
                // 将映射好的Dept对象作为Emp对象的属性设置进去
                emp.setDept(dept);

                // 每映射一行数据,生成一个Emp对象,添加到对应的集合中
                emps.add(emp);
            }
        } catch (DataAccessException e) {
            throw e;
        } catch (SQLException e) {
            throw new DataAccessException("数据访问异常");
        } finally {
            JDBCUtil.close(conn, ps, rs);
        }
        return emps;
    }

    /**
     * 一对多查询
     * 根据部门编号查询部门以及该部门下的所有员工信息
     * @return
     */
    public static Dept selectDeptById(Integer id) throws DataAccessException {
        Connection conn = null;
        PreparedStatement ps = null;
        ResultSet rs = null;
        Dept dept = null;
        try {
            conn = JDBCUtil.getConnection();
            String sql = new StringBuffer()
                    .append(" select d.id,d.name,d.remark,e.id 'e.id',e.name 'e.name',e.salary ")
                    .append(" from t_dept d ")
                    .append(" left join t_emp e ")
                    .append("   on d.id = e.dept_id ")
                    .append(" where d.id = ? ")
                    .toString();
            ps = conn.prepareStatement(sql);
            ps.setInt(1, id);
            rs = ps.executeQuery();
            while (rs.next()) {
                // 一对多映射
                // 当第一次进来的时候,映射部门对象
                // 后面再次进来,不再对部门进行映射
                // 当dept的值为null的时候,说明尚未映射过
                // 如果对dept进行映射,则dept的值不为null
                if (dept == null) {
                    dept = new Dept();
                    dept.setId(rs.getInt("id"));
                    dept.setName(rs.getString("name"));
                    dept.setRemark(rs.getString("remark"));
                }
                // 对于员工信息,每遍历一次,映射一个员工对象
                // 每次进来都需要对员工进行映射
                Emp emp = new Emp();
                emp.setId(rs.getInt("e.id"));
                emp.setName(rs.getString("e.name"));
                emp.setSalary(rs.getDouble("salary"));

                // 每次映射完成之后,将映射的员工对象添加到部门对象的对象集合属性中
                dept.addEmp(emp);
            }
            return dept;
        } catch (DataAccessException e) {
            throw e;
        } catch (SQLException e) {
            throw new DataAccessException("数据访问异常");
        } finally {
            JDBCUtil.close(conn, ps, rs);
        }
    }

}
